package com.ikcai.util;

import java.io.*;
import java.util.List;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
/**
 * Created by zhaoli on 2017/7/27.
 */
public class ImportExcle {

    /** 总行数 */
    private int		totalRows	= 0;
    /** 总列数 */
    private int		totalCells	= 0;
    /** 错误信息 */
    private String	errorInfo;

    /** 构造方法 */
    public ImportExcle()
    {
    }

    public int getTotalRows()
    {
        return totalRows;
    }

    public int getTotalCells()
    {
        return totalCells;
    }

    public String getErrorInfo()
    {
        return errorInfo;
    }

    public boolean validateExcel(String filePath)
    {
        /** 检查文件名是否为空或者是否是Excel格式的文件 */
        if (filePath == null || !(WDWUtil.isExcel2003(filePath) || WDWUtil.isExcel2007(filePath)))
        {
            errorInfo = "文件名不是excel格式";
            return false;
        }
        /** 检查文件是否存在 */
        File file = new File(filePath);
        if (file == null || !file.exists())
        {
            errorInfo = "文件不存在";
            return false;
        }
        return true;
    }

    public List<List<String>> read(String filePath)
    {
        List<List<String>> dataLst = new ArrayList<List<String>>();
        InputStream is = null;
        try
        {
            /** 验证文件是否合法 */
            if (!validateExcel(filePath))
            {
                System.out.println(errorInfo);
                return null;
            }
            /** 判断文件的类型，是2003还是2007 */
            boolean isExcel2003 = true;
            if (WDWUtil.isExcel2007(filePath))
            {
                isExcel2003 = false;
            }
            /** 调用本类提供的根据流读取的方法 */
            File file = new File(filePath);
            is = new FileInputStream(file);
            dataLst = read(is, isExcel2003);
            is.close();
        } catch (Exception ex)
        {
            ex.printStackTrace();
        } finally
        {
            if (is != null)
            {
                try
                {
                    is.close();
                } catch (IOException e)
                {
                    is = null;
                    e.printStackTrace();
                }
            }
        }
        /** 返回最后读取的结果 */
        return dataLst;
    }

    public List<List<String>> read(InputStream inputStream, boolean isExcel2003)
    {
        List<List<String>> dataLst = null;
        try
        {
            /** 根据版本选择创建Workbook的方式 */
            Workbook wb = null;
            if (isExcel2003)
            {
                wb = new HSSFWorkbook(inputStream);
            } else
            {
                wb = new XSSFWorkbook(inputStream);
            }
            dataLst = read(wb);
        } catch (IOException e)
        {
            e.printStackTrace();
        }
        return dataLst;
    }

    private List<List<String>> read(Workbook wb)
    {
        List<List<String>> dataLst = new ArrayList<List<String>>();
        /** 得到第一个shell */
        Sheet sheet = wb.getSheetAt(0);
        /** 得到Excel的行数 */
        this.totalRows = sheet.getPhysicalNumberOfRows();
        /** 得到Excel的列数 */
        if (this.totalRows >= 1 && sheet.getRow(0) != null)
        {
            this.totalCells = sheet.getRow(0).getPhysicalNumberOfCells();
        }
        /** 循环Excel的行 */
        for (int r = 0; r < this.totalRows; r++)
        {
            Row row = sheet.getRow(r);
            if (row == null)
            {
                continue;
            }
            List<String> rowLst = new ArrayList<String>();
            /** 循环Excel的列 */
            for (int c = 0; c < this.getTotalCells(); c++)
            {
                Cell cell = row.getCell(c);
                String cellValue = "";
                if (null != cell)
                {
                    // 以下是判断数据的类型
                    switch (cell.getCellType())
                    {
                        case HSSFCell.CELL_TYPE_NUMERIC: // 数字
                            cellValue = cell.getNumericCellValue() + "";
                            if (cellValue.indexOf(".") >= 0)
                            {
                                cellValue = cellValue.split("\\.")[0];
                            }
                            break;
                        case HSSFCell.CELL_TYPE_STRING: // 字符串
                            cellValue = cell.getStringCellValue();
                            break;
                        case HSSFCell.CELL_TYPE_BOOLEAN: // Boolean
                            cellValue = cell.getBooleanCellValue() + "";
                            break;
                        case HSSFCell.CELL_TYPE_FORMULA: // 公式
                            cellValue = cell.getCellFormula() + "";
                            break;
                        case HSSFCell.CELL_TYPE_BLANK: // 空值
                            cellValue = "";
                            break;
                        case HSSFCell.CELL_TYPE_ERROR: // 故障
                            cellValue = "非法字符";
                            break;
                        default:
                            cellValue = "未知类型";
                            break;
                    }
                }
                rowLst.add(cellValue);
            }
            /** 保存第r行的第c列 */
            dataLst.add(rowLst);
        }
        return dataLst;
    }

    public static void main(String[] args) throws Exception
    {
        ImportExcle poi = new ImportExcle();
        List<List<String>> list = poi.read("E:\\项目程序\\2017高校名单.xls");
        if (list != null)
        {
            int i = 0;
            if (list.size() <= 3)
            {
                return;
            }
            File file = new File("E:\\项目程序\\2017高校名单.sql");
            if (!file.exists())
            {
                file.createNewFile();
            }
            FileWriter fileWritter = new FileWriter(file, true);
            String province = "";
            try
            {
                for (i = 3; i < list.size(); i++)
                {
                    //System.out.print("第" + (i) + "行");
                    List<String> cellList = list.get(i);
                    if (cellList.get(1) == null || cellList.get(1).trim() == "")
                    {
                        String cell = cellList.get(0);
                        if (cell != null)
                        {
                            province = cell.split("（")[0];
                        }
                    } else
                    {
                        String sql = "INSERT INTO university(id,univname,univcode,authorities,city,grade,remark,vprepare,predate,province) values( ";
                        StringBuilder sb = new StringBuilder();
                        for (int j = 0; j < cellList.size(); j++)
                        {
                            if (j == 0)
                            {
                                sb.append(cellList.get(j));
                            } else
                            {
                                sb.append(",\"" + cellList.get(j) + "\"");
                            }
                        }
                        sql += sb.toString() + ",\"adm\",20170727152800,\"" + province + "\");";
                        System.out.println(sql);
                        fileWritter.write(sql);
                        fileWritter.write("\n");
                    }
                    if (i % 500 == 0)
                    {
                        fileWritter.flush();
                    }
                }
            } finally
            {
                if (fileWritter != null)
                {
                    fileWritter.close();
                }
            }
        }
    }
}
class WDWUtil
{
    public static boolean isExcel2003(String filePath)
    {
        return filePath.matches("^.+\\.(?i)(xls)$");
    }

    public static boolean isExcel2007(String filePath)
    {
        return filePath.matches("^.+\\.(?i)(xlsx)$");
    }
}
